op {
  graph_op_name: "BatchMatrixDiagPart"
  endpoint {
    name: "linalg.BatchMatrixDiagPart"
  }
}
